from langchain.prompts.prompt import PromptTemplate

from com.wp.langchain.exercise.LLMInit import LLMInit

llm = LLMInit().get_llm()

# 创建原始模板
template = "您是一位专业的程序员。\n对于信息 {text} 进行简短描述"
prompt = PromptTemplate(
    input_variables=["text"],
    template=template
)

# 输入提示
input = prompt.format(text="大模型 langchain")


# 得到模型的输出
output = llm.invoke(input)
# output = model.invoke("您是一位专业的程序员。对于信息 langchain 进行简短描述")

# 打印输出内容
print(output)